/**
 * 添加系统调用小例子
 * 此程序的作用是以模块的方式给系统添加自己的系统调用。
 */

#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/init.h>

#include <linux/unistd.h>
#include <linux/time.h>
#include <linux/uaccess.h>
#include <linux/sched.h>

#define QIAO 223  /*定义系统调用号*/
#define sys_call_table_add 0xc05fd260  /*系统调用表地址，可以通过/proc/kallsyms下找到*/

unsigned int clear_and_return_cr0(void);
void set_back_cr0(unsigned int val);

int orig_cr0;  /*用来保存初始cr0寄存器的值*/
unsigned long *sys_call_table = 0;  /*存放系统调用表的起始地址*/
static int (*func_point)(void);  /*保存QIAO这个号的系统调用在系统调用表中的内容*/

MODULE_LICENSE("GPL");
MODULE_AUTHOR("QIAO ZHAO");

unsigned int clear_and_return_cr0(void)
{
	unsigned int cr0 = 0;
	unsigned int ret;

	asm volatile("movl %%cr0, %%eax":"=a"(cr0));  /*读取cr0的值*/
	ret = cr0;

	cr0 &= 0xfffeffff;  /*清cr0中的第16位，这位是控制系统调用表是否可写，默认是不能写的*/
	asm volatile("movl %%eax, %%cr0"::"a"(cr0)); /*将修改后的值重新写入cr0中*/
	
	return ret;
}

void set_back_cr0(unsigned int val)
{
	asm volatile("movl %%eax, %%cr0"::"a"(val));
}

asmlinkage long sys_mycall(void)
{
	printk("mycall. current->pid is %d, current->comm is %s\n", current->pid, current->comm);
	return current->pid;
}

/*
asmlinkage long sys_mycall(int a)
{
	printk("%d, mycall. current->pid is %d, current->comm is %s\n", a, current->pid, current->comm);
	return current->pid;
}
*/

static int __init init_addsyscall(void)
{
	printk("start...\n");
	sys_call_table = (unsigned long *)sys_call_table_add;

	func_point = (int(*)(void))(sys_call_table[QIAO]);
	orig_cr0 = clear_and_return_cr0();
	sys_call_table[QIAO] = (unsigned long)&sys_mycall;  //将自己的系统调用函数写入sys_call_table表中
	set_back_cr0(orig_cr0);
	return 0;
}

static void __exit exit_addsyscall(void)
{
	orig_cr0 = clear_and_return_cr0();
	sys_call_table[QIAO] = (unsigned long)func_point;
	set_back_cr0(orig_cr0);
	printk("end...\n");
}

module_init(init_addsyscall);
module_exit(exit_addsyscall);
